
file(GLOB DEMO_TORCH "*.h" "*.hpp" "*.cpp")

add_library(demo_torch SHARED ${DEMO_TORCH})

# 指定链接语言为C++
set_target_properties(demo_torch PROPERTIES LINKER_LANGUAGE CXX)
target_link_libraries(demo_torch ${TORCH_LIBRARIES})



add_library(libchap4 MLP.cpp CNN.cpp)
target_link_libraries(libchap4 ${TORCH_LIBRARIES})

add_library(libchap5 Lenet5.cpp)
target_link_libraries(libchap5 ${TORCH_LIBRARIES})

